'''
---记账本管理系统（Json版）-----
项目需求：
包含以下功能：
1.初始化记账本数据
    时间                      项目      分类    金额
    2021/8/21 09:00:00      工资      收入      20000
    2021/8/21 09:00:00      聚餐      支出      -200
2.读取json数据
3.写入json数据
4.显示账单（显示总支出金额，总收入金额，结余金额
5.写入账单（时间按照格式显示
'''
# 4.显示账单（显示总支出金额，总收入金额，结余金额
import time
import orm


def show_book():
    income = 0
    spend = 0
    jsonpath = r'book.json'
    data = orm.read_data(jsonpath)
    # print(data_discuz)
    for bill in data:
        money = float(bill['金额'])
        print(bill)
        if money > 0:
            income += money
        else:
            spend -= money
    print(f'总收入金额为:{income}')
    print(f'总支出金额为:{spend}')
    print(f'结余金额为：{income - spend}')


# 5.写入账单（时间按照格式显示
def write_book():
    path = r'book.json'
    time1 = time.strftime('%Y-%m-%d %H:%M:%S')
    # print(time1)
    project = input('请输入你的记账项目：')
    category = input('请输入你的记账项目分类(收入/支出)：')
    money = input('请输入金额：')
    new_book = {"时间": time1, "项目": project, "分类": category, "金额": money}
    book_list = orm.read_data(path)
    book_list.append(new_book)
    orm.wirte_data(book_list, path)
    print('添加成功')


if __name__ == '__main__':
    # show_book()
    write_book()
